package com.learning.rocketmq.producer;

import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.client.producer.MessageQueueSelector;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.common.message.Message;
import org.apache.rocketmq.common.message.MessageQueue;
import org.apache.rocketmq.remoting.common.RemotingHelper;

import java.util.List;

/**
 * ClassName: CusProducer
 * Description: 自定义消息发送规则 MessageQueueSelector
 * Date: 2018/12/27 11:07 【需求编号】
 *
 * @author Sam Sho
 * @version V1.0.0
 */
public class CusProducer {

    /**
     * 一个Topic 会有多个Message Queue ，如果使用Producer 的默认配置，这
     * 个Producer 会轮流向各个Message Queue 发送消息。Consumer 在消费消息的
     * 时候，会根据负载均衡策略，消费被分配到的Message Queue ，如果不经过特
     * 定的设置，某条消息被发往哪个Message Queue ，被哪个Consumer 消费是未
     * 知的。
     * <p>
     * 如果业务需要我们把消息发送到指定的Message Queue 里，比如把同一
     * 类型的消息都发往相同的Message Queue ， 该怎么办呢？ 可以用 MessageQueueSelector
     *
     * @param args
     */
    public static void main(String[] args) throws Exception {

        // 设置 producerGroup
        DefaultMQProducer producer = new DefaultMQProducer("base_producer_group");
        // 设置 nameServer
        producer.setNamesrvAddr("10.0.64.106:9876;10.0.64.107:9876");
        producer.start();

        for (int i = 0; i < 100; i++) {
            Message message = new Message();
            message.setTopic("base_topic");
            message.setTags("base_api");
            message.setBody(("Hello RocketMQ > " + i).getBytes(RemotingHelper.DEFAULT_CHARSET));


            SendResult result = producer.send(message, new OrderMessageQueueSelector(), 13);
            System.out.printf("%s%n", result);
        }

        producer.shutdown();
    }

    /**
     * 自定义规则
     */
    public static class OrderMessageQueueSelector implements MessageQueueSelector {
        @Override
        public MessageQueue select(List<MessageQueue> mqs, Message msg, Object orderKey) {
            int id = Integer.parseInt(orderKey.toString());
            int idMainIndex = id / 100;
            int size = mqs.size();
            int index = idMainIndex % size;
            return mqs.get(index);
        }
    }
}
